import torch
from Data_TIMIT import Data_TIMIT

overlap = 64
data_tr = Data_TIMIT('test', overlap = overlap)
data_te = Data_TIMIT('train', mix_num = 1)

test_loader = torch.utils.data.DataLoader(data_tr, batch_size = 1, shuffle = True, num_workers = 4)

train_loader = torch.utils.data.DataLoader(data_te, batch_size = 50, shuffle = True, num_workers = 4)


# torch.save(train_loader, 'data/half_win_train_mul.pth')
# torch.save(test_loader, 'data/half_win_test_mul.pth')
